package com.atlige.gzmall.order.config;


import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;

/**
 * @author Jeremy Li
 * @data 2021/4/1 - 23:52
 */
@Configuration
public class MyRabbitConfig {
    @Autowired
    RabbitTemplate rabbitTemplate;

    @Bean
    public MessageConverter messageConverter(){
        return new Jackson2JsonMessageConverter();
    }

    /**
     * 定制RabbitTemplate
     * 1、服务器收到消息就回调
     *      1、# 开启发送端确认 spring.rabbitmq.publisher-confirms: true
     *      2、设置确认回调
     * 2、消息正确抵达队列进行回调
     *      1、 publisher-returns: true
     *          template.mandatory: true
     *      2、设置确认回调ReturnCallback
     * 3、消费端确认（保证每个消息被正确消费，此时才可以broker删除这个消息）
     *      1、默认是自动确认机制，只要消息接收到，服务端就会删除这个消息
     *      问题：
     *          我们收到很多消息，自动回复给服务器ACK，只有一个消息处理成功，宕机了。发送消息丢失
     *          手动确认    # 手动ack消息
     *     listener.simple.acknowledge-mode: manual
     *     只要没有明确告诉MQ消息被签收，没有ACK，消息就一直是unacked状态，即使Consumer宕机，消息也不会丢失，会重新变为ready状态
     *     下次有新的consumer连接进来就发给他
     */
    @PostConstruct // MyRabbitConfig对象创建完成以后，执行这个方法
    public void initRabbitTemplate(){
        // 设置确认回调
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * 1、只要消息抵达Broker就ack=true
             * @param correlationData 当前消息的唯一关联数据（消息的唯一id）
             * @param ack 消息是否成功收到
             * @param cause 失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                /**
                 * 保证可靠消息
                 * 1、做好消息确认机制(publisher、consumer[手动ACK])
                 * 2、每个发送的消息在数据库做好记录，定期将失败的消息重新发送
                 */
                // 服务器收到消息
                System.out.println("confirm...correlationData[" + correlationData +
                        "] --> ack[" + ack + "] --> causer[" + cause + "]");
            }
        });
        // 设置消息抵达队列的确认回调
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 只有消息没有投递给指定的队列，就触发此失败回调
             * @param message 投递失败的消息详情
             * @param replyCode 回复的状态码
             * @param replyText 回复的文本内容
             * @param exchange 当时这个消息发送给哪个交换机
             * @param routingKey 当时这个消息发送给哪个路由键
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                // 报错误了，修改数据库当前消息的状态 -> 错误
                System.out.println("Fail Message [" + message + "] --> replyCode [" + replyCode
                 + "] --> replyText [" + replyText + "] --> exchange [" + exchange + "] --> routingKey [" + routingKey + "]");
            }
        });
    }

}
